# -*- coding: utf-8 -*-

# Define your item pipelines here
#
# Don't forget to add your pipeline to the ITEM_PIPELINES setting
# See: https://doc.scrapy.org/en/latest/topics/item-pipeline.html
import pymysql
from kaola.items import KaolaArticleItem,KaolaItem


class KaolaPipeline(object):
    def open_spider(self,spider):
        self.coon = pymysql.connect('127.0.0.1','root','123456','kaola',charset = 'utf8')
        self.cursor = self.coon.cursor()
    def process_item(self, item, spider):

        try:
            if isinstance(item,KaolaItem):
                sql,data = item.save_to_mysql()

                self.cursor.execute(sql,data)

                self.coon.commit()

                print('---------作者信息写入成功---------')
            elif isinstance(item,KaolaArticleItem):

                sql,data = item.save_to_mysql()

                self.cursor.execute(sql,data)

                self.coon.commit()

                print('----------文章信息写入成功----------')
            return item
        except Exception as e:

            print('------写入失败原因为%s-----'%e)


    def close_spider(self,spider):
        self.cursor.close()
        self.coon.close()
